package com.linzm.leetcode.primary.exercises1_20230114;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @Author zimingl
 * @Date 2023/1/15 19:44
 * @Description: 拥有最多糖果的孩子
 */
public class Demo23_1431 {
    /**
     * 输入：candies = [2,3,5,1,3], extraCandies = 3
     * 输出：[true,true,true,false,true]
     * 解释：
     * 孩子 1 有 2 个糖果，如果他得到所有额外的糖果（3个），那么他总共有 5 个糖果，他将成为拥有最多糖果的孩子。
     * 孩子 2 有 3 个糖果，如果他得到至少 2 个额外糖果，那么他将成为拥有最多糖果的孩子。
     * 孩子 3 有 5 个糖果，他已经是拥有最多糖果的孩子。
     * 孩子 4 有 1 个糖果，即使他得到所有额外的糖果，他也只有 4 个糖果，无法成为拥有糖果最多的孩子。
     * 孩子 5 有 3 个糖果，如果他得到至少 2 个额外糖果，那么他将成为拥有最多糖果的孩子。
     */
    public static void main(String[] args) {
        Demo23_1431 demo23_1431 = new Demo23_1431();
        int[] candies = new int[]{2,3,5,1,3};
        int extraCandies = 3;
        List<Boolean> list = demo23_1431.kidsWithCandies(candies, extraCandies);
        String[] strings = list.stream().map(String::valueOf).toArray(String[]::new);
        System.out.println(String.join(" ", strings));
    }

    private List<Boolean> kidsWithCandies3(int[] candies, int extraCandies) {
        int max = Arrays.stream(candies).max().getAsInt();
        List<Boolean> list = new ArrayList<>();
        for (int i = 0; i < candies.length; i++) {
            if (candies[i] + extraCandies >= max) {
                list.add(Boolean.TRUE);
            } else {
                list.add(Boolean.FALSE);
            }
        }
        return list;
    }

    private List<Boolean> kidsWithCandies(int[] candies, int extraCandies) {
        int[] ints = Arrays.copyOf(candies, candies.length);
        Arrays.sort(ints);
        int max = ints[candies.length - 1];
        List<Boolean> list = new ArrayList<>();
        for (int i = 0; i < candies.length; i++) {
            if (candies[i] + extraCandies >= max) {
                list.add(Boolean.TRUE);
            } else {
                list.add(Boolean.FALSE);
            }
        }
        return list;
    }

    public List<Boolean> kidsWithCandies2(int[] candies, int extraCandies) {
        int n = candies.length;
        int maxCandies = 0;
        for (int i = 0; i < n; ++i) {
            maxCandies = Math.max(maxCandies, candies[i]);
        }
        List<Boolean> ret = new ArrayList<>();
        for (int i = 0; i < n; ++i) {
            ret.add(candies[i] + extraCandies >= maxCandies);
        }
        return ret;
    }
}
